import matplotlib.pyplot as plt
import imageio
import random
#plt.hold(False)
file = open("submission.csv", "r").read().splitlines()[1:]
start = 30
stop = min(start+10, len(file)-1)
lines = file[start:stop]
for line in lines:
l, r = line.split(',')
plt.figure(figsize=(3,3))
query_path = 'test'+'/'+l+'.jpg'
images = ['index'+'/'+i+'.jpg' for i in r.split()]
# Show the query image
plt.imshow(imageio.imread(query_path))
plt.figure(figsize=(20,10))
# Then show 4 of the similar images
columns = 5
for i, image in enumerate(images):
plt.subplot(len(images) / columns + 1, columns, i + 1)
plt.imshow(imageio.imread(image))